package io.gitee.luoyuan800.tool.llm.bean.config;

import lombok.Data;

import java.util.Map;

@Data
public class AiConfig {
    private String defaultModel;
    Map<String, AiAgentConfig> agentConfigs;
    Map<String, ModelConfig> models;

    public AiAgentConfig getAgentConfig(String name) {
        AiAgentConfig agentConfig;
        if (agentConfigs.containsKey(name)) {
            agentConfig = agentConfigs.get(name);
            if (agentConfig.getModelConfig() == null) {
                agentConfig.setModelConfig(models.get(agentConfig.getModel()));
            }
        } else {
            agentConfig = new AiAgentConfig();
            agentConfig.setModel(defaultModel);
            agentConfig.setModelConfig(models.get(defaultModel));
        }
        return agentConfig;
    }
}
